package com.tianyouxiu.springai.repository.impl;

import com.tianyouxiu.springai.repository.ChatHistoryRepository;
import org.springframework.stereotype.Component;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Component
public class InMemoryChatHistoryRepository implements ChatHistoryRepository {
  private static final Map<String, List<String>> chatHistory = new HashMap<>();

  @Override
  public void save(String type, String chatId) {
    /*    if (!chatHistory.containsKey(type)) {
      chatHistory.put(type, new ArrayList<>());
    }
    List<String> chatIds = chatHistory.get(type);*/

    List<String> chatIds = chatHistory.computeIfAbsent(type, k -> new ArrayList<>());
    if (chatIds.contains(chatId)) {
      return;
    }
    chatIds.add(chatId);
  }

  @Override
  public List<String> getChatIds(String type) {
    /* List<String> chatIds = chatHistory.get(type);
    return chatIds == null ? List.of() : chatIds;*/
    return chatHistory.getOrDefault(type, List.of());
  }
}
